import torch
import os

# 基础路径
ROOT_DIRECTORY = os.path.dirname(os.path.realpath('./'))

# 文本文件夹
file_dir = [
   './assets/sql'
]
# 定义持久化路径
Persist_dir = "./DB"

# embedding model
EMBEDDING_MODEL_NAME = "./model/embedding_model/paraphrase-multilingual-MiniLM-L12-v2"

EMBEDDING_DEVICE = "cpu" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"

# 推理大模型
MODEL_NAME = "./model/push_model/Qwen-7B-Chat-Int8/qwen/Qwen-7B-Chat-Int8/"